# 定义对象库层的基类
from selenium.webdriver.support.wait import WebDriverWait

from utils import UtilsDriver


class BasePage:
    def __init__(self):
        self.driver = UtilsDriver.get_ttmp_driver()  # 获取自媒体平台的驱动

    # 定义获取元素对象的方法
    def get_element(self, location):
        """
        :param location: 表示的是元素的定位信息
        :return:  返回的是一个元素对象
        """
        wait = WebDriverWait(self.driver, 10)
        return wait.until(lambda x:x.find_element(*location))


# 封装操作层基类
class BaseHandle:
    # 定义针对元素输入的操作方法
    def input_text(self, element, text):
        """
        :param element: 表示的是元素对象
        :param text:  表示输入的文本内容
        :return:
        """
        element.clear()
        element.send_keys(text)